package edu.hit.crawler.util;

import java.security.MessageDigest;

/**
 * MD5的算法在RFC1321 中定义
 * 在RFC 1321中，给出了Test suite用来检验你的实现是否正确： 
 * MD5 ("") = d41d8cd98f00b204e9800998ecf8427e 
 * MD5 ("a") = 0cc175b9c0f1b6a831c399e269772661 
 * MD5 ("abc") = 900150983cd24fb0d6963f7d28e17f72 
 * MD5 ("message digest") = f96b697d7cb7938d525a2f31aaf161d0 
 * MD5 ("abcdefghijklmnopqrstuvwxyz") = c3fcd3d76192e4007dfb496cca67e13b 
 * 
 * @author 
 *
 * 传入参数：一个字符串
 * 传出参数：字节数组的 MD5 结果字符串
 */
public class GenerateMD5 {
	public static String getMD5(byte[] source) {
		
		String s = null;
		char hexDigits[] = {       // 用来将字节转换成 16 进制表示的字符
		     '0', '1', '2', '3', '4', '5', '6', '7', '8', '9', 'a', 'b', 'c', 'd',  'e', 'f'}; 
		
		try {
			
			MessageDigest md = MessageDigest.getInstance( "MD5" );
			md.update( source );
			
		    byte tmp[] = md.digest();          // MD5 的计算结果是一个 128 位的长整数，用字节表示就是 16 个字节
		    char str[] = new char[16 * 2];     //每个字节用 16 进制表示的话，使用两个字符，所以表示成 16 进制需要 32 个字符		                                              
		    int k = 0;                         // 表示转换结果中对应的字符位置
		    
		    for (int i = 0; i < 16; i++) {     // 从第一个字节开始，对 MD5 的每一个字节转换成 16 进制字符的转换
		    	byte bt = tmp[i];           // 取第 i 个字节取字节中高 4 位的数字转换, >>> 为逻辑右移，将符号位一起右移
		    	str[k++] = hexDigits[bt >>> 4 & 0xf];  
		    	str[k++] = hexDigits[bt & 0xf];       // 取字节中低 4 位的数字转换
		    } 
		    
		    s = new String(str);                        // 换后的结果转换为字符串
		 }catch( Exception e ) {
		    System.out.println(e.getMessage());
		    return "";
		 }
		 
		 return s;
    }
	
	/**
	 * 将传入的字符串经过MD5签名
	 * @param source 传入的字符串
	 * @return	返回签名后的MD5信息（固定长度为32byte）
	 */
	public static String getMD5(String source) { 
		if(source.length() != 0) {
			return getMD5(source.getBytes());
		} else {
			return "";
		}
	}
	
	/**
	 * 将传入的字符串经过MD5签名
	 * @param source 传入的字符串
	 * @return	返回签名后的MD5信息（固定长度为32byte） Integer类型
	 */
	public static int getMD5Code(String source) {
		int res = 0;
		/*String md5 =getMD5(source).substring(source.length() - 4, source.length()); 
		if(md5.length() != 0) {
			res = Integer.parseInt(md5, 16);
		}*/ 
		
		if(source.length() != 0) {
			source = getMD5(source);
			try {
				int sum = 0;
				for(int i = 0; i < source.length(); i++) {
					sum +=  Integer.parseInt(String.valueOf(source.charAt(i)), 16);
				}
				res = sum;
			}catch(Exception e) {
				System.out.println("error:" + e.getMessage());
			}
		}
		
		return res;
	}
}
